package com.xiaolin.oa.utils;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.TypeReference;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Objects;
import java.util.concurrent.TimeUnit;

/**
 * @author xiaolin
 * @description 描述
 * @since 2024/8/6
 */
@Component
public class RedisUtil {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    /**
     * 判断key是否存在
     *
     * @param key 键
     * @return true 存在 false不存在
     */
    public boolean hasKey(String key) {
        if (StringUtils.isBlank(key)) {
            return false;
        }
        return redisTemplate.hasKey(key);
    }

    /**
     * 根据value从一个set中查询,是否存在
     *
     * @param key   键
     * @param value 值
     * @return true 存在 false不存在
     */
    public boolean sHasKey(String key, Object value) {
        if (StringUtils.isBlank(key) || !redisTemplate.hasKey(key)) {
            return false;
        }
        return redisTemplate.opsForSet().isMember(key, value);
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key    键
     * @param value  值
     * @param second 时间(秒) time要大于0 如果time小于等于0 将设置无限期
     * @return true成功 false 失败
     */
    public boolean set(String key, Object value, long second) {
        if (Objects.isNull(value)) {
            return false;
        }
        if (value instanceof String) {
            redisTemplate.opsForValue().set(key, (String) value, second, TimeUnit.SECONDS);
        } else {
            redisTemplate.opsForValue().set(key, JSON.toJSONString(value), second, TimeUnit.SECONDS);
        }
        return true;
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key      键
     * @param value    值
     * @param time     时间
     * @param timeUnit 单位
     * @return true成功 false 失败
     */
    public boolean set(String key, Object value, long time, TimeUnit timeUnit) {
        if (Objects.isNull(value)) {
            return false;
        }
        if (value instanceof String) {
            redisTemplate.opsForValue().set(key, (String) value, time, timeUnit);
        } else {
            redisTemplate.opsForValue().set(key, JSON.toJSONString(value), time, timeUnit);
        }
        return true;
    }

    /**
     * 普通缓存放入
     *
     * @param key   键
     * @param value 值
     * @return true成功 false失败
     */
    public boolean set(String key, Object value) {
        if (Objects.isNull(value)) {
            return false;
        }
        if (value instanceof String) {
            redisTemplate.opsForValue().set(key, (String) value);
        } else {
            redisTemplate.opsForValue().set(key, JSON.toJSONString(value));
        }
        return true;
    }

    /**
     * 将数据放入set缓存
     *
     * @param key    键
     * @param values 值 可以是多个
     * @return 成功个数
     */
    public long sSet(String key, String... values) {
        try {
            return redisTemplate.opsForSet().add(key, values);
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 普通缓存获取
     *
     * @param key 键
     * @return 值
     */
    public String get(String key) {
        if (StringUtils.isBlank(key)) {
            return null;
        }
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 对象缓存获取
     *
     * @param key 键
     * @return 值
     */
    public <T> T get(String key, Class<T> tClass) {
        if (StringUtils.isBlank(key) || !redisTemplate.hasKey(key)) {
            return null;
        }
        return JSON.parseObject(redisTemplate.opsForValue().get(key), tClass);
    }

    /**
     * 列表缓存获取
     *
     * @param key 键
     * @return 值
     */
    public <T> T get(String key, TypeReference<T> t) {
        if (StringUtils.isBlank(key) || !redisTemplate.hasKey(key)) {
            return null;
        }
        return JSON.parseObject(redisTemplate.opsForValue().get(key), t);
    }

    /**
     * 删除缓存
     *
     * @param key
     * @return
     */
    public boolean del(String key) {
        return redisTemplate.delete(key);
    }


    /**
     * 指定缓存失效时间
     *
     * @param key    键
     * @param second 时间(秒)
     * @return
     */
    public boolean expire(String key, long second) {
        if (second > 0 && redisTemplate.hasKey(key)) {
            return redisTemplate.expire(key, second, TimeUnit.SECONDS);
        }
        return false;
    }


}
